import pandas as pd
import os

# --------------------------
# 用户配置部分
# --------------------------

# 翻译文件所在目录
translation_dir = r'C:\Users\admin\Desktop\5.6语言处理\第二次处理\翻译文件'

# 需要处理的语言列表（必须和第一步拆分时一致）
languages = [
    '意大利语', '法语', '日语', '德语', '西班牙语',
    '俄语', '葡萄牙语', '阿拉伯语', '繁体', '简体'
]

# --------------------------
# 自动遍历每个语言文件进行处理
# --------------------------

for lang in languages:
    file_path = os.path.join(translation_dir, f'{lang}.xlsx')
    
    if not os.path.exists(file_path):
        print(f"⚠️ 文件不存在: {file_path}，跳过该语言")
        continue

    # 读取翻译文件
    df = pd.read_excel(file_path)

    modified_col = f'{lang}-修改后'
    original_col = f'{lang}-未修改'

    if modified_col not in df.columns:
        print(f"⚠️ 文件 {file_path} 缺少 '{modified_col}' 列，跳过该语言")
        continue

    if original_col not in df.columns:
        print(f"⚠️ 文件 {file_path} 缺少 '{original_col}' 列，跳过该语言")
        continue

    # 填充空值：如果“修改后”为空，则使用“未修改”的值
    df[modified_col] = df[modified_col].fillna(df[original_col])

    # 保存回原文件（或可选保存为新文件）
    df.to_excel(file_path, index=False)
    print(f"✅ 已处理并更新文件: {file_path}")